import torch.nn as nn

# 定义一个继承自 torch.nn.Module 的类
class Net(nn.Module):
    # 初始化方法
    def __init__(self):
        # 调用父类的初始化方法
        super().__init__()
        # 定义第一层为线性层，输入维度为 1，输出维度为 2
        self.layer1 = nn.Linear(1, 2)
        # 定义第二层为线性层，输入维度为 2（与上一层的输出维度相同），输出维度为 1
        self.layer2 = nn.Linear(2, 1)

    # 定义前向传播方法
    def forward(self, x):
        # 将输入 x 通过第一层，并使用 relu 激活函数处理输出
        x = nn.functional.relu(self.layer1(x))
        # 将上一步的输出通过第二层，得到最终预测结果
        y2 = self.layer2(x)
        # 返回预测结果
        return y2
